import gradio as gr
from main import detect_ai_text

# 创建Gradio界面
interface = gr.Interface(
    fn=detect_ai_text,
    inputs=[gr.Textbox(label="文本内容")],
    outputs=[
        gr.Label(label="检测结果"),
    ],
    title="AI文本检测器",
    description="输入文本，检测是否可能为AI生成。",
)

# 启动界面
interface.launch(server_port=7860)
